python3 prof_demo.py --device npu:0 --amp --opt adamw